from django import forms
from .models import File, Folder


class FileUploadForm(forms.ModelForm):
    class Meta:
        model = File
        fields = ['name', 'file']


class FolderCreateForm(forms.ModelForm):
    class Meta:
        model = Folder
        fields = ['name']
